import numpy as np
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import *

import matplotlib as mpl

x, y = np.mgrid[-1: 1: 20j, -1: 1: 20j]
z = x ** 2 + y ** 2

print(x)
print(y)

figure = plt.figure()
ax = figure.add_subplot(111, projection='3d')

ax.plot_surface(x, y, z, rstride=4, cstride=4)

ax.view_init(45, 45)

plt.show()
